Assets/Scripts/Imported/Tilemap/Brushes/Line Brush/Scripts/Editor/LineBrush.cs

   1 using System;
   2 using System.Collections;
   3 using System.Collections.Generic;
   4 using UnityEngine;
   5 using UnityEngine.Tilemaps;
   6 using System.Linq;
   7
   8 namespace UnityEditor
   9 {
  10     [CustomGridBrush(true, false, false, "Line Brush")]
  11     public class LineBrush : GridBrush
  12     {
  13         public bool lineStartActive = false;
  14         public bool fillGaps = false;
  15         public Vector3Int lineStart = Vector3Int.zero;
  16
  17         public override void Paint(GridLayout grid, GameObject brushTarget, Vector3Int position)
  18         {
  19             if (lineStartActive)
  20             {
  21                 Vector2Int startPos = new Vector2Int(lineStart.x, lineStart.y);
  22                 Vector2Int endPos = new Vector2Int(position.x, position.y);
  23                 if (startPos == endPos)
  24                     base.Paint(grid, brushTarget, position);
  25                 else
  26                 {
  27                     foreach (var point in GetPointsOnLine(startPos, endPos, fillGaps))
  28                     {
  29                         Vector3Int paintPos = new Vector3Int(point.x, point.y, position.z);
  30                         base.Paint(grid, brushTarget, paintPos);
  31                     }
  32                 }
  33                 lineStartActive = false;
  34             }
  35             else
  36             {
  37                 lineStart = position;
  38                 lineStartActive = true;
  39             }
  40         }
  41
  42         [MenuItem("Assets/Create/Line Brush")]
  43         public static void CreateBrush()
  44         {
  45             string path = EditorUtility.SaveFilePanelInProject("Save Line Brush", "New Line Brush", "asset", "Save Line Brush", "Assets");
  46
  47             if (path == "")
  48                 return;
  49
  50             AssetDatabase.CreateAsset(ScriptableObject.CreateInstance<LineBrush>(), path);
  51         }
  52
  53         /// <summary>
  54         /// Added option to fill gaps for continuous lines.
  55         /// </summary>
  56         public static IEnumerable<Vector2Int> GetPointsOnLine(Vector2Int startPos, Vector2Int endPos, bool fillGaps)
  57         {
  58             var points = GetPointsOnLine(startPos, endPos);
  59             if (fillGaps)
  60             {
  61                 var rise = endPos.y - startPos.y;
  62                 var run = endPos.x - startPos.x;
  63
  64                 if (rise != 0 || run != 0)
  65                 {
  66                     var extraStart = startPos;
  67                     var extraEnd = endPos;
  68
  69
  70                     if (Mathf.Abs(rise) >= Mathf.Abs(run))
  71                     {
  72                         // up
  73                         if (rise > 0)
  74                         {
  75                             extraStart.y += 1;
  76                             extraEnd.y += 1;
  77                         }
  78                         // down
  79                         else // rise < 0
  80                         {
  81
  82                             extraStart.y -= 1;
  83                             extraEnd.y -= 1;
  84                         }
  85                     }
  86                     else // Mathf.Abs(rise) < Mathf.Abs(run)
  87                     {
  88
  89                         // right
  90                         if (run > 0)
  91                         {
  92                             extraStart.x += 1;
  93                             extraEnd.x += 1;
  94                         }
  95                         // left
  96                         else // run < 0
  97                         {
  98                             extraStart.x -= 1;
  99                             extraEnd.x -= 1;
 100                         }
 101                     }
 102
 103                     var extraPoints = GetPointsOnLine(extraStart, extraEnd);
 104                     extraPoints = extraPoints.Except(new[] { extraEnd });
 105                     points = points.Union(extraPoints);
 106                 }
 107
 108             }
 109
 110             return points;
 111         }
 112
 113         // http://ericw.ca/notes/bresenhams-line-algorithm-in-csharp.html
 114         public static IEnumerable<Vector2Int> GetPointsOnLine(Vector2Int p1, Vector2Int p2)
 115         {
 116             int x0 = p1.x;
 117             int y0 = p1.y;
 118             int x1 = p2.x;
 119             int y1 = p2.y;
 120
 121             bool steep = Math.Abs(y1 - y0) > Math.Abs(x1 - x0);
 122             if (steep)
 123             {
 124                 int t;
 125                 t = x0; // swap x0 and y0
 126                 x0 = y0;
 127                 y0 = t;
 128                 t = x1; // swap x1 and y1
 129                 x1 = y1;
 130                 y1 = t;
 131             }
 132             if (x0 > x1)
 133             {
 134                 int t;
 135                 t = x0; // swap x0 and x1
 136                 x0 = x1;
 137                 x1 = t;
 138                 t = y0; // swap y0 and y1
 139                 y0 = y1;
 140                 y1 = t;
 141             }
 142             int dx = x1 - x0;
 143             int dy = Math.Abs(y1 - y0);
 144             int error = dx / 2;
 145             int ystep = (y0 < y1) ? 1 : -1;
 146             int y = y0;
 147             for (int x = x0; x <= x1; x++)
 148             {
 149                 yield return new Vector2Int((steep ? y : x), (steep ? x : y));
 150                 error = error - dy;
 151                 if (error < 0)
 152                 {
 153                     y += ystep;
 154                     error += dx;
 155                 }
 156             }
 157             yield break;
 158         }
 159     }
 160
 161     [CustomEditor(typeof(LineBrush))]
 162     public class LineBrushEditor : GridBrushEditor
 163     {
 164         private LineBrush lineBrush { get { return target as LineBrush; } }
 165         private Tilemap lastTilemap;
 166
 167         public override void OnPaintSceneGUI(GridLayout grid, GameObject brushTarget, BoundsInt position, GridBrushBase.Tool tool, bool executing)
 168         {
 169             base.OnPaintSceneGUI(grid, brushTarget, position, tool, executing);
 170             if (lineBrush.lineStartActive)
 171             {
 172                 Tilemap tilemap = brushTarget.GetComponent<Tilemap>();
 173                 if (tilemap != null)
 174                     lastTilemap = tilemap;
 175
 176                 // Draw preview tiles for tilemap
 177                 Vector2Int startPos = new Vector2Int(lineBrush.lineStart.x, lineBrush.lineStart.y);
 178                 Vector2Int endPos = new Vector2Int(position.x, position.y);
 179                 if (startPos == endPos)
 180                     PaintPreview(grid, brushTarget, position.min);
 181                 else
 182                 {
 183                     foreach (var point in LineBrush.GetPointsOnLine(startPos, endPos, lineBrush.fillGaps))
 184                     {
 185                         Vector3Int paintPos = new Vector3Int(point.x, point.y, position.z);
 186                         PaintPreview(grid, brushTarget, paintPos);
 187                     }
 188                 }
 189
 190                 if (Event.current.type == EventType.Repaint)
 191                 {
 192                     var min = lineBrush.lineStart;
 193                     var max = lineBrush.lineStart + position.size;
 194
 195                     // Draws a box on the picked starting position
 196                     GL.PushMatrix();
 197                     GL.MultMatrix(GUI.matrix);
 198                     GL.Begin(GL.LINES);
 199                     Handles.color = Color.blue;
 200                     Handles.DrawLine(new Vector3(min.x, min.y, min.z), new Vector3(max.x, min.y, min.z));
 201                     Handles.DrawLine(new Vector3(max.x, min.y, min.z), new Vector3(max.x, max.y, min.z));
 202                     Handles.DrawLine(new Vector3(max.x, max.y, min.z), new Vector3(min.x, max.y, min.z));
 203                     Handles.DrawLine(new Vector3(min.x, max.y, min.z), new Vector3(min.x, min.y, min.z));
 204                     GL.End();
 205                     GL.PopMatrix();
 206                 }
 207             }
 208         }
 209
 210         public override void ClearPreview()
 211         {
 212             base.ClearPreview();
 213             if (lastTilemap != null)
 214             {
 215                 lastTilemap.ClearAllEditorPreviewTiles();
 216                 lastTilemap = null;
 217             }
 218         }
 219     }
 220 }